//===----------------------------------------------------------*- swift -*-===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2016 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See http://swift.org/LICENSE.txt for license information
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

%import gyb

% for Self in ['SKShapeNode', 'SKSpriteNode', 'SKTextureAtlas', 'SKTexture']:

extension ${Self} : CustomPlaygroundQuickLookable {
  public var customPlaygroundQuickLook: PlaygroundQuickLook {
    // this code comes straight from the quicklooks
      
    let data = (self as AnyObject)._copyImageData?()
    // we could send a Raw, but I don't want to make a copy of the
    // bytes for no good reason make an NSImage out of them and
    // send that
#if os(OSX)
    if let data = data {
      return .sprite(NSImage(data: data))
    } else {
      return .sprite(NSImage())
    }
#elseif os(iOS) || os(watchOS) || os(tvOS)
    if let data = data {
      return .sprite(UIImage(data: data))
    } else {
      return .sprite(UIImage())
    }
#endif
  }
}
